#!/bin/bash

# 目标 MongoDB 连接参数
AUTH_DB="admin"

# 手动输入 MongoDB 用户名和密码
echo "请输入 MongoDB HOST:"
read HOST
echo "请输入 MongoDB PORT:"
read PORT
echo "请输入 MongoDB 用户:"
read USERNAME
echo "请输入 MongoDB 密码:"
read -s PASSWORD


# 解压缩 tar.gz 文件
echo "解压缩 mongo_dump.tar.gz..."
tar -xzf mongo_dump.tar.gz -C /tmp

# 检查解压后的目录
if [ ! -d "/tmp/mongo_dump" ]; then
  echo "解压后的目录不存在, 请检查 tar.gz 文件是否正确"
  exit 1
fi

# 遍历每个文件夹，进行 mongorestore 导入
for folder in /tmp/mongo_dump/*/; do
  # 提取文件夹名字
  folder_name=$(basename "$folder")
  echo "正在导入数据库: $folder_name"

  # 执行 mongorestore
  mongorestore --host $HOST --port $PORT --username $USERNAME --password $PASSWORD --authenticationDatabase $AUTH_DB --db $folder_name "/tmp/mongo_dump/$folder_name"
  
  if [ $? -eq 0 ]; then
    echo "$folder_name 导入成功!"
  else
    echo "$folder_name 导入失败!"
  fi
done

echo "所有数据库导入完成!"
